#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @FileName  :1.4 udp聊天器.py
# @Time      :2020/7/9 15:51
# @Author    :亮亮
# @说明       :可以收发数据的聊天器
# @总结       :
import socket


def main():
    # 创建socket套接字
    udp_data = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    # 绑定端口和IP
    local_addr = ('', 9090)
    udp_data.bind(local_addr)

    # 输入发送方的IP和端口
    dest_ip = input('输入对方IP:')
    dest_port = int(input('输入对方端口:'))

    # 创建循环
    while True:
        # 输入发送的内容
        sent_data = input('输入要发送的内容:')

        # 发送
        udp_data.sendto(sent_data.encode('utf-8'), (dest_ip, dest_port))

        # 接收显示
        recv_data = udp_data.recvfrom(1024)
        print(recv_data[0].decode('utf-8'))

        # 判断结束对话
        if recv_data[0].decode('utf-8') == 'close':
            break

    # 关闭套接字/程序
    udp_data.close()


if __name__ == '__main__':
    main()
